package 快乐数;

import java.util.HashSet;
import java.util.Set;

public class Solution {
    /*
    编写一个算法来判断一个数 n 是不是快乐数。
    「快乐数」 定义为：
    对于一个正整数，每一次将该数替换为它每个位置上的数字的平方和。
    然后重复这个过程直到这个数变为 1，也可能是 无限循环 但始终变不到 1。
    如果这个过程 结果为 1，那么这个数就是快乐数。
    如果 n 是 快乐数 就返回 true ；不是，则返回 false 。
     */

    public boolean isHappy(int n) {
        // 数学家和计算机科学家通过计算和理论分析已经证明，对于任何初始数字，数字的平方和序列要么达到1，要么最终陷入循环

        // 使用 哈希表 法，以此记录是否出现了重复的数字
        Set<Integer> set = new HashSet<>();
        // 对于任何初始数字，数字的平方和序列要么达到1，要么最终陷入循环
        while ( n != 1 && !set.contains(n) ){
            // 将数字加入到集合中
            set.add(n);
            // 计算新的n值
            n = getNewNum(n);
        }
        return n == 1;
    }

    public int getNewNum(int n){
        int result = 0;
        while ( n != 0){
            int num = n % 10;
            result += num*num;
            n /= 10;
        }
        return result;
    }

}
